package com.csx.linkedlist;

/**
 * 链表
 * @author 陈胤训
 * <p>
 * create: 2019-04-02 15:53
 **/
public class LinkedList<E> {

    private class Node{

        /** 值  */
        public E e;

        /** 下一个节点  */
        public Node next;

        public Node(E e,Node next){
            this.e = e;
            this.next = next;
        }

        public Node(E e){
            this(e,null);
        }

        public Node(){
            this(null,null);
        }

        @Override
        public String toString(){
            return e.toString();
        }
    }

    /** 虚拟的头节点  */
    private Node dummyHead;

    /**  大小 */
    int size;

    public LinkedList(){
        dummyHead = new Node(null,null);
        size = 0;
    }

    /** 获取链表中的元素个数  */
    public int getSize(){
        return size;
    }

    /** 返回链表是否为空  */
    public boolean isEmpty(){
        return size == 0;
    }

    /** 在链表头添加新的元素 e  */
    public void addFirst(E e){
//        Node node =  new Node(e);
//        node.next = head;
//        head = node;
//        dummyHead = new Node(e,dummyHead);
//        size ++;
        add(0,e);
    }

    /** 添加  */
    public void add(int index, E e){
        if (index < 0 || index > size){
            throw new IllegalArgumentException("add 方法 . index位置有误 ");
        }
            Node prev = dummyHead;
            for (int i = 0; i < index ; i++) {
                prev = prev.next;
            }
//            Node node = new Node(e);
//            node.next = prev.next;
//            prev.next = node;
            /**  prev.next   改的是 dummyHead 的值*/
            prev.next = new Node(e,prev.next);
            size ++;
        }


    /** 在链表末尾添加新元素  */
    public void addLast(E e){
        add(size,e);
    }

    /** 获取链表的第index个位置的元素  */
    /** 在链表中不是一个常用的操作  */
    public E get(int index){
        if (index < 0 || index > size){
            throw new IllegalArgumentException("get 方法 . index位置有误 ");
        }
        Node cur = dummyHead.next;
        for (int i = 0 ; i < index ; i ++){
            cur = cur.next;
        }
        return cur.e;
    }

    /** 获得链表的第一个元素  */
    public E getFirst(){
        return get(0);
    }

    /** 获得链表的最后一个元素  */
    public E getLast(){
        return get(size - 1);
    }

    /** 修改链表的第index个位置的元素  */
    /** 在链表中不是一个常用的操作  */
    public void set(int index , E e){
        if (index < 0 || index > size){
            throw new IllegalArgumentException("set 方法 . index位置有误 ");
        }
        Node cur = dummyHead.next;
        for (int i = 0; i < index; i++) {
            cur = cur.next;
        }
        cur.e = e;
    }

    /** 查找链表中是否有元素e  */
    public boolean contains(E e){
        Node cur = dummyHead.next;
        while(cur != null){
            if (cur.e.equals(e)){
                return true;
            }
            cur =cur.next;
        }
        return false;
    }

    /** 删除链表的第index个位置的元素  */
    /** 在链表中不是一个常用的操作  */
    public E remove(int index){
        if (index < 0 || index > size){
            throw new IllegalArgumentException("remove 方法 . index位置有误 ");
        }
        Node prev = dummyHead;
        for (int i = 0; i < index; i++) {
            prev = prev.next;
        }
        /** 要删除的链表  */
        Node retNode = prev.next;
        prev.next = retNode.next;
        retNode.next = null;
        retNode.e = null;
        size --;
        return retNode.e;
    }

    /** 删除链表的第一个元素  */
    public E removeFirst(){
        return remove(0);
    }

    /** 删除链表的最后一个元素  */
    public E removeLast(){
        return remove(size - 1);
    }

    // 从链表中删除元素e
    public void removeElement(E e){

        Node prev = dummyHead;
        while(prev.next != null){
            if(prev.next.e.equals(e))
                break;
            prev = prev.next;
        }

        if(prev.next != null){
            Node delNode = prev.next;
            prev.next = delNode.next;
            delNode.next = null;
            size --;
        }
    }

    @Override
    public String toString() {
        StringBuilder res = new StringBuilder();

//        Node cur = dummyHead.next;
//        while(cur != null){
//            res.append(cur + "->");
//            cur =cur.next;
//        }
        for(Node cur = dummyHead.next;cur != null ;cur = cur.next){
            res.append(cur + "->");
        }
        res.append("NULL");
        return res.toString();
    }

    public static void main(String[] args) {
        LinkedList<Integer> list = new LinkedList<>();
        for (int i = 0; i < 5; i++) {
         list.addFirst(i);
            System.out.println(list);
        }
        list.add(2,9);
        list.remove(1);
        System.out.println(list);

    }
}
